package com.gwcloud.leetcode.tags.backtrace;

import java.util.ArrayList;
import java.util.List;

/** 
* @ClassName   : Leetcode131 
* @author      : ganwei 
* @date        : 2020年5月9日 
* @Description : TODO 
*/
public class Leetcode0131 {
	List<List<String>> result = new ArrayList<>();
	List<List<Integer>> resultdot = new ArrayList<>();
	/**
	 * -解题思路，继续练练我的回溯大法
	 */
    public List<List<String>> partition(String s) {
    	for(int j = 0; j < s.length(); j++) {
    		backtrack(s.length(), new ArrayList<Integer>(), j, 1);
    	}
    	/**
    	 * 根据resultdot生成result
    	 */
    	for(List<Integer> l : resultdot) {
    		List<String> rl = new ArrayList<String>();
    		int rn = l.size();
    		int lindex = -1;
    		int rindex = 0;
    		while(rindex <= rn) {
    			int lv,rv;
    			if(lindex == -1) {
    				lv = 0;
    			}else {
    				lv = l.get(lindex);
    			}
    			
    			if(rindex == rn) {
    				rv = s.length();
    			}else {
    				rv = l.get(rindex);
    			}
    			
    			String ts = s.substring(lv, rv);
    			
    			if(!isPalindrome(ts)) {
    				break;
    			}
    			rl.add(s.substring(lv, rv));
    			lindex++;
    			rindex++;
    			
    		}
    		
    		if(rl.size() == l.size()+1) {
    			result.add(rl);
    		}
    		
    		
    	}
    	return result;
    }
    /**
     * -回溯主体
     */
	public void backtrack(int n, List<Integer> li, int k, int index){
		if(li.size() == k){
			resultdot.add(new ArrayList<Integer>(li));
			return;
		}
		for(int i = index; i < n; i++){
			li.add(i);
			
			backtrack(n,li, k,i+1);
			li.remove(li.size()-1);
		}
	}
    /**
     * -判断回文字符串
     */
    public boolean isPalindrome(String s) {
    	if(s.length()==1) {
    		return true;
    	}
    	int lindex = 0;
    	int rindex = s.length()-1;
    	while(lindex < rindex) {
    		if(s.charAt(lindex) != s.charAt(rindex)) {
    			return false;
    		}
    		lindex++;
    		rindex--;
    	}
    	
    	return true;
    }
    /**
     * -别人家的回溯打法怎么可能那么简单。。。
     */
    public List<List<String>> partition_1(String s){
    	if(s.length() == 0) {
    		return result;
    	}
    	
    	backtrack_1(s,0,new ArrayList<String>());
    	
    	return result;
    }
    
    /**
     * -回溯主体
     */
    public void backtrack_1(String s,int start, List<String> sl) {
    	if(start == s.length()) {
    		result.add(new ArrayList<String>(sl));
    		return;
    	}
    	
    	for(int i = start; i < s.length(); i++) {
    		if(!isPalindrome(s.substring(start, i+1))) {
    			continue;
    		}
    		sl.add(s.substring(start,i+1));
    		backtrack_1(s, i+1, sl);
    		sl.remove(sl.size()-1);
    	}
    	
    }
    
    public static void main(String[] args) {
    	Leetcode0131 l131 = new Leetcode0131();
    	l131.partition_1("aab");
    	for(List<String> l : l131.result) {
    		for(String ii : l) {
    			System.out.print(ii);
    			System.out.print(' ');
    		}
    		System.out.println();
    	}
    	
    }
}
